Implement And Train Vit From Scratch For Image Recognition - Pytorch